// 打家劫舍2

package Leetcode;

import java.util.Arrays;

// 错误
class Solution_213_1 {
    public int rob(int[] nums) {
        if (nums.length == 1)
            return nums[0];
        if (nums.length == 2)
            return Math.max(nums[0], nums[1]);
        if (nums.length == 3)
            return Math.max(nums[0], Math.max(nums[1], nums[2]));
        int[] dp = new int[nums.length];
        boolean[] isUse = new boolean[nums.length]; // 是否使用第一个元素
        dp[0] = nums[0];
        dp[1] = nums[1];
        dp[2] = nums[0] + nums[2];
        isUse[0] = true;
        isUse[1] = false;
        isUse[2] = true;

        // 最后一位另作讨论，因为存在环
        for (int i = 3; i < nums.length - 1; i++) {
            if (dp[i - 2] > dp[i - 3]) {
                dp[i] = dp[i - 2] + nums[i];
                isUse[i] = isUse[i - 2];
            } else if (dp[i - 2] < dp[i - 3]) {
                dp[i] = dp[i - 3] + nums[i];
                isUse[i] = isUse[i - 3];
            } else { // 当相等时我们尽量选择没有使用第一个元素的序列
                if (isUse[i - 2] == false) {
                    dp[i] = dp[i - 2] + nums[i];
                    isUse[i] = isUse[i - 2];
                    continue;
                }
                dp[i] = dp[i - 3] + nums[i];
                isUse[i] = isUse[i - 3];
            }
        }

        // 如果到最后一位却用到了第一位则需要减去，因为是环
        int value1 = dp[nums.length - 3] + nums[nums.length - 1];
        int value2 = dp[nums.length - 4] + nums[nums.length - 1];
        if (isUse[nums.length - 3] == true)
            value1 -= Math.min(nums[0], nums[nums.length - 1]);
        if (isUse[nums.length - 4] == true)
            value2 -= Math.min(nums[0], nums[nums.length - 1]);
        dp[nums.length - 1] = Math.max(value1, value2);

        return Math.max(dp[nums.length - 2], dp[nums.length - 1]);
    }
}

class Solution_213_2 {
    public int rob(int[] nums) {
        if (nums.length == 1)
            return nums[0];
        if (nums.length == 2)
            return Math.max(nums[0], nums[1]);
        if (nums.length == 3)
            return Math.max(nums[0], Math.max(nums[1], nums[2]));
        return Math.max(rob198(Arrays.copyOfRange(nums, 1, nums.length)),
                rob198(Arrays.copyOfRange(nums, 0, nums.length - 1)));
    }

    public int rob198(int[] nums) { // 这是第198题的代码
        if (nums.length == 1)
            return nums[0];
        if (nums.length == 2)
            return Math.max(nums[0], nums[1]);
        if (nums.length == 3)
            return Math.max(nums[0] + nums[2], nums[1]);
        int[] dp = new int[nums.length];
        dp[0] = nums[0];
        dp[1] = nums[1];
        dp[2] = nums[0] + nums[2];
        for (int i = 3; i < nums.length; i++)
            dp[i] = Math.max(dp[i - 2], dp[i - 3]) + nums[i];

        return Math.max(dp[nums.length - 2], dp[nums.length - 1]);
    }
}
